# 导包
import unittest

import app
from api.login import IHRMLogin

# 创建测试类
class LoginAPI(unittest.TestCase):
# 前置方法
    def setUp(self):
        self.login_api = IHRMLogin()
# 后置方法
#     无session不需要关闭
# 创建测试用方法

    def test01_case001(self):
        response = self.login_api.login({"mobile": "13800000002", "password": "123456"})
        print(response.json())
        self.assertEqual(200, response.status_code)
        self.assertEqual(True, response.json().get("success"))
        self.assertEqual(10000, response.json().get("code"))
        self.assertIn("操作成功", response.json().get("message"))
        # 断言

        # 提取token
        app.TOKEN = "Bearer " + response.json().get("data")
        print(app.TOKEN)
        app.header_data["Authorization"] = app.TOKEN
        print(app.TOKEN)